updated for cucumber test