Got unit test working again