small refactor update