fixup! Use return_dict instead of return_list.