Put logic in test instead of model