fixed a small error in tests