add missing cleanup step