Fixed error in tests