Fix the cucumber test