Adds forgotten cleanup