Only mock the checkRegistrationCode method