cleanup
fix test that was broken