added missing line that caused test failings