I know what you mean, when you actually sit down to play, it's so easy to get distracted and start playing other stuff. What helped me is to actually write down the speed or bpm and write down how many times I played it at that speed next to it.
Maybe that will help you too.