another fix for test code.