Fixed unit test registration so that test passes