diff --git a/privacy/bolton/models.py b/privacy/bolton/models.py
index 06d1c4b..7503157 100644
--- a/privacy/bolton/models.py
+++ b/privacy/bolton/models.py
@@ -60,7 +60,7 @@ class BoltonModel(Model):
     self._layers_instantiated = False
     self._dtype = dtype
 
-  def call(self, inputs, training=False):  # pylint: disable=arguments-differ
+  def call(self, inputs):  # pylint: disable=arguments-differ
     """Forward pass of network
 
     Args:
diff --git a/tutorials/bolton_tutorial.ipynb b/tutorials/bolton_tutorial.ipynb
index b60e612..f682592 100644
--- a/tutorials/bolton_tutorial.ipynb
+++ b/tutorials/bolton_tutorial.ipynb
@@ -321,7 +321,7 @@
     "                                                  kernel_regularizer=reg_layer\n",
     "                                                  )\n",
     "        \n",
-    "    def call(self, inputs, training=False):\n",
+    "    def call(self, inputs):\n",
     "        return self.output_layer(inputs)\n",
     "\n",
     "optimizer = tf.optimizers.SGD()\n",
@@ -420,13 +420,13 @@
   "pycharm": {
    "stem_cell": {
     "cell_type": "raw",
+    "source": [],
     "metadata": {
      "collapsed": false
-    },
-    "source": []
+    }
    }
   }
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}
\ No newline at end of file