Fixed a unit test failure

Resulted from a bad interface implementation