Make script work on TeamCity as well