code cleanup
small bugfix