Even in this situation it’s uneven. If the company pays you the right amount within a couple weeks, nothing happens. It’s as if they never shorted you.
If you take the money from the company, you - at least - pay an additional fine. And/or go to prison. The company doesn’t have either consequence for attempting to steal from you.