fixed test breakage