mini fix for test
clean up test