Fix bug with heights on reload and closed #3