Fixing last unit test reference, build passes locally