The code is currently only working for 3-point water. The original implementation was also tested for TIP-4P, so there must be a regression somewhere.