Fixed a test case that occasionally failed