Fixup type assignment error in test