Fix for registration of model