Small fix for test that was failing in TeamCity