Fix height bug