Bug fix for height calc.