Added missing change to test suite setup