Fixed a small bug in tests.