Added missing cleanup decorator import.
Merge pull request #2718 from ajdawson/animation-test-fail

Added missing cleanup decorator import.