fixes bug in importing custom models in previous commit