Fixed a small error in tests