[TEST] Fixed failing unit test.