Fixing height bug