Fix bug that comment shows in wrong place