fixed test, seems to pass