fixed failing tests.

was looking in the wrong file