Tattvartha Sutra says that the state of mind and inner disposition at death influences the next birth.

Exact text:

> बह्वारम्भपरिग्रहत्वं नारकस्यायुषः ॥१५॥ > Excessive infliction of pain and excessive attachment to possessions cause birth in hell. ( jainqq.org)

> माया तैर्यग्योनस्य ॥१६॥ > Deceitfulness causes birth in the animal and plant world. ( jainqq.org)

> अल्पारम्भपरिग्रहत्वं मानुषस्य ॥१७॥ > Slight activity and slight attachment to possessions cause human birth. ( jainqq.org)

In the explanation of these sutras, the text specifically mentions the condition at the time of death:

> black thought-colouration (leśyā) with cruel concentration (raudradhyāna) at the time of death leads to hell-birth, and > blue or grey thought-colouration with sorrowful concentration (ārtadhyāna) at the time of death leads to animal/plant birth. ( jainqq.org)

It also says that for human birth, one should be free from distressful disposition at the time of death. ( jainqq.org)

If you want, I can also give the Romanized Sanskrit and a line-by-line meaning of these sutras.