Fix tests, thanks jhass! [ci skip]