take out test code
fix test code