Fix test case. Broken in last commit.