I have reduced the time step and things seem to be working quite well