diff --git a/hungarian.go b/hungarian.go
index 6d6dabc..e87af16 100644
--- a/hungarian.go
+++ b/hungarian.go
@@ -231,6 +231,8 @@ func SolveMax(matrix [][]float64) map[int]map[int]float64 {
 
 	b.removeExtra()
 
+	b.checkAndReplace()
+
 	return b.ReducedExtremums
 }