Use .cleanup() for tests