Fix crash fix

Forgot to import updater module