Switched model to fix login