Improve updater tests