Creo que seria mejor: pensat en beure aigua
Antes de infinitivo la preposición "en" se convierte en "a".