Fix munkres import check