Minor.
Add missing login checks