Fix error causing test breakage