Fixed unit tests after adding login