cleanup; small fix