diff --git a/odl/discr/partition.py b/odl/discr/partition.py
index c92388c8882..fc62408fc4f 100644
--- a/odl/discr/partition.py
+++ b/odl/discr/partition.py
@@ -897,8 +897,7 @@ def uniform_partition_fromintv(intv_prod, shape, nodes_on_bdry=False):
     array([ 0.2,  0.6,  1. ])
     """
 
-    grid = uniform_grid_fromintv(intv_prod, shape,
-                                     nodes_on_bdry=nodes_on_bdry)
+    grid = uniform_grid_fromintv(intv_prod, shape, nodes_on_bdry=nodes_on_bdry)
 
     return RectPartition(intv_prod, grid)
 
diff --git a/odl/test/discr/partition_test.py b/odl/test/discr/partition_test.py
index c6690261ba5..44c9927856a 100644
--- a/odl/test/discr/partition_test.py
+++ b/odl/test/discr/partition_test.py
@@ -15,6 +15,8 @@
 # You should have received a copy of the GNU General Public License
 # along with ODL.  If not, see <http://www.gnu.org/licenses/>.
 
+from __future__ import division
+
 import pytest
 import numpy as np