Updated the test to work with teamcity