SetUp test fix