Try to fix the custom-header function