Fixed error with model submodule