Fix incorrect assertion in test