Update test directory in before() function