Added missing cleanup decorator import.

I'm seeing a test error because the `@cleanup` decorator is used in test_animation.py but is not imported. The combination of these two commits looks to be the problem: 61bce5ddc20cdef650a6cf2c5ff05898a665ea04 and 03c71f53cd8d1d39f4ef16687800ce07a3bea06b.