Changed to IModelCustom, crashing atm