Added fix to #256 height problem