Fix login function