Fix tests (aka 99th commit for 2 line change)