I recall Rabbi Avogdor Miller ??”? mentioning that if one worked on oneself and it becomes second nature (e.g. always getting up for davening on time) then one still gets rewarded for doing it as it if were difficult.
One does not get “punished” for making it second nature.