Minor bug fix in cucumber tests