The two guys were enjoying a nice swim