Fix for incorrect height