You can book the boat in both directions if you like, rather than returning by coach, although I seem to remember having to do it as two separate bookings. I saw the waterwheels by doing a land-based trip the day before -- I got the train to Pasłęk, from where it was a bit more than an hour's...