fix test
clean up