Fix diff_main return type