fixed test to match source updates