Fixed failing unit test from yesterday