Fix the bug with the height problem.