Плавание. Собака не возвращается на берег

Доброго времени  суток уважаемые Сергей и Елена Смирновы!

У нас  ам.стаф. кобель, 1 год 1 мес. посещал кинолога, знает команды (рядом, сидеть, лежать, апорт, место).  У нас возникла неожиданная проблема с собакой.

Все было впорядке, пока собака не увидела воду, первые два раза он купался на 10 м. поводке в реке, не на какие команды, любимые игрушки он не реагировал, приходилось его насильно вытягивать поводком.

В третий раз просто без поводка подошли к воде (пруд) и собака поплыла куда-то, проплыв 120 м. он начал плавать кругами не реагируя ни на что так продолжалось 20 мин., пришлось сыну раздеваться и плыть за ним, в результате чего, чуть двое не утонули.

Уже заказали собаке спас. жилет,  но все таки возникает вопрос, что с этим делать и как научить собаку по команде возвращаться на берег или плыть рядом.

С уважением Виктория