Fixed the shaking bug, even though it was totally awesome