fixed test, matched code to test