Fixing a test code bug.