Fix: a bug in handling returning types.