Fixes to login module