updated tests after refactoring the user flow