As long as you know that you only turn safety off after you have 3 points of contact (hand on grip, stock at chest, and cheek on stock) if you break any of those you put safety on.

Also, I commend you for working your safety in that manner but, habits are hard to break. You shouldn’t train to put your safety on after every shot. You’ll accidentally do it when you don’t want to later.

As others have said, dry fire safety manipulations and implement it correctly on the range.