Oh yes, the gravity code is a little whacky. Your fix seems perfectly fine to me though.

"Falls das Resultat nicht einfach nur dermassen gut aussieht, sollten Sie nochmal von vorn anfangen..." - Manual

Check out my new game: Pogostuck: Rage With Your Friends