Assign model class to loader when it is loaded