minor
small code update