diff --git a/element_moseq/moseq_train.py b/element_moseq/moseq_train.py
index a6b6949..2830a8f 100644
--- a/element_moseq/moseq_train.py
+++ b/element_moseq/moseq_train.py
@@ -227,7 +227,7 @@ def make(self, key):
 
         if pose_estimation_method == "deeplabcut":
             setup_project(
-                project_dir = kpms_project_output_dir.as_posix(),
+                project_dir=kpms_project_output_dir.as_posix(),
                 deeplabcut_config=(kpset_dir / "config.yaml")
                 or (kpset_dir / "config.yml"),
             )
diff --git a/notebooks/tutorial.ipynb b/notebooks/tutorial.ipynb
index e4fefc5..e0ee380 100644
--- a/notebooks/tutorial.ipynb
+++ b/notebooks/tutorial.ipynb
@@ -3614,7 +3614,7 @@
     "full_fit_key = {\n",
     "    **pca_task_key,\n",
     "    \"full_latent_dim\": 4,\n",
-    "    \"full_kappa\": 10000.,\n",
+    "    \"full_kappa\": 10000.0,\n",
     "    \"full_num_iterations\": 5,\n",
     "    \"full_fit_desc\": \"Fitting task with kappa = 10000 ms\",\n",
     "}\n",
@@ -3638,7 +3638,7 @@
     "full_fitting_key_2 = {\n",
     "    **pca_task_key,\n",
     "    \"full_latent_dim\": 4,\n",
-    "    \"full_kappa\": 5000.,\n",
+    "    \"full_kappa\": 5000.0,\n",
     "    \"full_num_iterations\": 5,\n",
     "    \"full_fit_desc\": \"Fitting task with kappa = 5000 ms\",\n",
     "}\n",