Fixed unit test references