fixed test to pass