Fixed issue with header not getting updated