Minor refactor in snapshot worker