Fixed a small bug in one of the tests.