add back some tests