Fix tests
Allow slightly different reg flow for teachers and students