Starting model class tests
Try to fix test