Is it permissible for a Muslim to say: "O Allah, reward me in my affliction and compensate me with something better than it," if the affliction was a result of a forbidden act he committed?

Sins can be a cause for calamities to befall, and a servant must repent from them. Allah Almighty has stated that turning away from His remembrance leads to a straitened life, and that corruption on land and sea is due to the hands of people. The story of the people of the garden in Surat Al-Qalam shows that they were deprived of their garden because of their intention to deprive the poor. When they repented and sought forgiveness, Allah replaced it with something better for them. The Prophet, peace and blessings be upon him, mentioned that whoever is afflicted by a calamity and says: "Indeed we belong to Allah, and indeed to Him we will return. O Allah, reward me in my calamity and replace it with something better for me," Allah will reward him and replace it with something better for him. The word "calamity" includes everything that troubles a believer.