Fix to set headers correctly.