[Layers] Update and correct missing custom code

Change-Id: I8407a2d4324c28bf63802938652d5dd5fe52ab0d
Signed-off-by: Quentin Le Menez <quentin.lemenez@cea.fr>
2 files changed