Add test cleanup decorator