fix error in tests