Fix shake animation