updated tests to fix test bug