A quick fix for better animation