Fixing issue with the heights