Spiel im Wasser - tauchen

Hallo, unser Hund steht total darauf, Steine aus dem Wasser zu holen. Das ist entstanden durch unsere erste HuTa, die Besitzerin war mit den Hunden immer im Freilauf unterwegs und wenn sie irgendwo am Wasser mit den Tieren war, hat sie Steine geworfen und die Hunde haben dann im Wasser getobt und sind geschwommen. Seitdem ist er total auf Steine fixiert. Man kann mit ihm am Wasser spielen, er schwimmt auch schön, aber er liebt es richtig mit dem Kopf unter Wasser zu gehen und holt dann teils richtig große Steine raus. Teilweise bleibt er von sich aus richtig lang mit dem Kopf unter Wasser. Wenn es mal nur vereinzelt wäre hätten wir kein Problem damit. Abgesehen davon, dass es ihn irgendwann frustriert, habe ich auch gesundheitliche Bedenken...vor allem in Bezug auf mögliche Wassermengen, die er wahrscheinlich schluckt. Kann man ihm das wieder abtrainieren? Freue mich über Anregungen Vielen Dank und ein schönes Wochenende