fix bug at test code