diff --git a/pytorch/lovasz_losses.py b/pytorch/lovasz_losses.py
index de6fff6..89a993d 100644
--- a/pytorch/lovasz_losses.py
+++ b/pytorch/lovasz_losses.py
@@ -12,7 +12,7 @@
 try:
     from itertools import  ifilterfalse
 except ImportError: # py3k
-    from itertools import  filterfalse
+    from itertools import  filterfalse as ifilterfalse
 
 
 def lovasz_grad(gt_sorted):