fix test units