diff --git a/kaffe/tensorflow/network.py b/kaffe/tensorflow/network.py
index 6f3b153..6351091 100644
--- a/kaffe/tensorflow/network.py
+++ b/kaffe/tensorflow/network.py
@@ -130,11 +130,11 @@ def conv(self,
                 output = convolve(input, kernel)
             else:
                 # Split the input into groups and then convolve each of them independently
-                input_groups = tf.split(3, group, input)
-                kernel_groups = tf.split(3, group, kernel)
+                input_groups = tf.split(input,group, 3)
+                kernel_groups = tf.split(kernel, group, 3)
                 output_groups = [convolve(i, k) for i, k in zip(input_groups, kernel_groups)]
                 # Concatenate the groups
-                output = tf.concat(3, output_groups)
+                output = tf.concat(output_groups, 3)
             # Add the biases
             if biased:
                 biases = self.make_var('biases', [c_o])
@@ -177,7 +177,7 @@ def lrn(self, input, radius, alpha, beta, name, bias=1.0):
 
     @layer
     def concat(self, inputs, axis, name):
-        return tf.concat(concat_dim=axis, values=inputs, name=name)
+        return tf.concat(values=inputs, concat_dim=axis,  name=name)
 
     @layer
     def add(self, inputs, name):