Ana de Armas is relaxing in the water