update tests

Yep, nice call. Can't believe I missed that.