You have the right idea, but remember that tax rate is not 100%. So if a company gets a fine on X dollar, that reduces their surplus by X, and they pay e.g X*20% less in tax (if the tax rate is 20%).