Fix broken test when running full suite of tests