try to fix test code