Fix tests to run from setup command