Practice returning manageable serves toward a broad in-bounds area, review direction and depth separately, and add court movement only after the basic return is understandable.
Give beginner serve practice a broad, recognizable landing goal inside the correct service court, with legal motion checked separately and no loose targets in playing space.
Use a slow, predictable feed to notice where you lose track of the ball before responding to missed contact by swinging harder or adding more movement.