small refactorinigs
add missing line