Fixed bug from failing test.