The dress was fun to work with in the water