Forgot to run specs after refactoring