Update expected types in assertions