Handling of custom models initialisation

Part of the initialisation was not applied.