fix for test bug