Fix improperly set-up test case