minor fixes
cleaning up the test