update type comment
fix test