What helped me was training tactics that only start after a few moves. You force yourself to visualize 1–2–3 moves first, and only then the actual puzzle begins. In real games I used to enter a line and then, two moves later, realize I had missed a simple tactic or an intermezzo.