small fix for test code