diff --git a/keras_contrib/callbacks/dead_relu_detector.py b/keras_contrib/callbacks/dead_relu_detector.py
index f8f6839..2cfe37b 100644
--- a/keras_contrib/callbacks/dead_relu_detector.py
+++ b/keras_contrib/callbacks/dead_relu_detector.py
@@ -78,7 +78,7 @@ class DeadReluDetector(Callback):
 
             dead_neurons = np.sum(np.sum(activation_values, axis=axis) == 0)
 
-            dead_neurons_share = dead_neurons / total_featuremaps
+            dead_neurons_share = float(dead_neurons) / float(total_featuremaps)
             if (self.verbose and dead_neurons > 0) or dead_neurons_share >= self.dead_neurons_share_threshold:
                 str_warning = 'Layer {} (#{}) has {} dead neurons ({:.2%})!'.format(layer_name, layer_index,
                                                                                     dead_neurons, dead_neurons_share)
diff --git a/tests/keras_contrib/callbacks/dead_relu_detector_test.py b/tests/keras_contrib/callbacks/dead_relu_detector_test.py
index 84f1b91..e4a1783 100644
--- a/tests/keras_contrib/callbacks/dead_relu_detector_test.py
+++ b/tests/keras_contrib/callbacks/dead_relu_detector_test.py
@@ -1,7 +1,11 @@
 import pytest
 import numpy as np
 import sys
-import io
+
+if (sys.version_info > (3, 0)):
+    from io import StringIO
+else:
+    from StringIO import StringIO
 
 from keras_contrib import callbacks
 from keras.models import Sequential
@@ -17,8 +21,10 @@ def check_print(do_train, expected_warnings, nr_dead=None, perc_dead=None):
     :param nr_dead: int
     :param perc_dead: float, 10% should be written as 0.1
     """
+    
     saved_stdout = sys.stdout
-    out = io.StringIO()
+
+    out = StringIO()
     sys.stdout = out    # overwrite current stdout
 
     do_train()
@@ -133,7 +139,7 @@ def test_DeadDeadReluDetector_conv():
     shape_weights = (5, 5, 4, n_out)
     shape_out = (n_samples, n_out)
 
-    def do_test(weights_bias, expected_warnings, verbose, nr_dead = None, perc_dead = None):
+    def do_test(weights_bias, expected_warnings, verbose, nr_dead=None, perc_dead=None):
         """
         :param perc_dead: as float, 10% should be written as 0.1
         """