Fix wiggly ball bug