Fixes #5

Probably needs porting to the new updater framework too ?
Merge pull request #6 from tomchiverton/patch-1

Fixes #5