Посоветуйте что делать

Мы в год начали ходить в бассейн, ребенок в восторге. Занятия проходит в игровой форме - накорми бегемотика, забрось мячик в корзину и т.п. По периметру бассейна стоят ящики с различными развивающими материалами, мы очень любили эти ящики переворачивать, но я объяснила что это делать не надо и Тимур вроде бы понял. Теперь у него новая "фишка" - бегать на четвереньках вокруг бассейна, с огромным восторгом, после чего он готов дальше плавать (сам спускается в воду). В итоге большую часть времени мы находимся "на суше". Пыталась держать на ручках(изолировать), объяснять, не пускать, отвлекать, но не помогает. Может ненадолго отвлечься - принести бегемотику морковку и опять лезет на сушу. Как поступать? Настаивать на своем? Как? Или пусть ползает?