Fix test directory identification