Fix login bugs