Auth improvements in tests