Adding the changes to login tests