Fixed the models loading