diff --git a/vehicle-detection/darkflow/net/ops/convolution.py b/vehicle-detection/darkflow/net/ops/convolution.py
index 09eac6b..554b933 100644
--- a/vehicle-detection/darkflow/net/ops/convolution.py
+++ b/vehicle-detection/darkflow/net/ops/convolution.py
@@ -85,8 +85,8 @@ def batchnorm(self, layer, inp):
                 'updates_collections' : None,
                 'is_training': layer.h['is_training']
                 })
-            v = tf.__version__.split('.')[1]
-            if int(v) < 12: key = 'initializers'
+            v = tf.__version__
+            if int(v) < 0.12: key = 'initializers'
             else: key = 'param_initializers'
             args.update({key : layer.w})
             return slim.batch_norm(inp, **args)