Restoring running `test`

- Tests now work from fix in previous commit