Fix tests

CI would be nice... nah... don't care...