Fix for retrieving model class