fix #2

Fix https://github.com/openworm/Smoothed-Particle-Hydrodynamics/issues/2