fix the cleanup