Fixed bug with height