added more assertions to make sure we're getting right test data