"IRS imputes extra tax on those benefits and CA is not supposed to"

I'm not sure I can be more help. You are not taxed a second time on a benefit you already paid for out of post-tax dollars, which is how you describe that $4k. Did you enter it somewhere other than the W2 input?