reverting tests

Ran the wrong tests so reverting to the correct ones.