Remove dead code from tests