Add back necessary headers