Fix up the test so it is testing correctly