Adds fix for heat directory bug