Fixed failing tests from a dumb oversight