minor
Fixed clean-up