Fixed a failing unit test