fix test problem