Fixing broken models test