update tests

Great, thanks. Fixing.