You were being robbed by Ballas. They pointed guns at you, a gunfight started, however at least one kept his gun pointed at you to stop you from running. At that point you still have to fear for your life. You proceeded to run away, which lead to Ballas shooting you.
In my opinion the punishment is right, but of course this is up to my higherups.