Fix for login controller