Cleanup for test modules