Fixing failing Cucumber feature