updated test to cleanup