fixed test to match updated code