BUG: Fixed failing unit test.