fix test to be like official test