fix test module bug