Fixing cucumber test (whoops...)