some code cleanup
minor change