- fix test code
