diff --git a/pyproject.toml b/pyproject.toml
index bed9c7cd6f7174ad1ef183d204c2e3cf080e0bc8..0320fd51c4403eb14b9fc83478962f6b2b40d92a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,10 +12,10 @@ python = "^3.8,<3.11"
 h5py = "3.1.0"
 pandas = "1.3.3"
 PyYAML = "^6.0"
-torchvision = "^0.12.0"
+torchvision = "0.14.1"
 attrdict = "^2.0.1"
 tqdm = "^4.64.0"
-torch = "^1.11.0"
+torch = "1.13.1"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
diff --git a/src/behavior_model/data/larva_dataset.py b/src/behavior_model/data/larva_dataset.py
index 1140518724721e3e4112f0539405bce8ac1c0385..ae35bd30522e9bdbcea705c64f3f0da860837ac2 100644
--- a/src/behavior_model/data/larva_dataset.py
+++ b/src/behavior_model/data/larva_dataset.py
@@ -81,7 +81,7 @@ class LarvaDataset(Dataset):
 def import_label(s):
     s = s.decode('utf-8') if isinstance(s, bytes) else s
     # MaggotUBA only understands BACK, BEND, HUNCH, ROLL, RUN, STOP.
-    # With the `small_motion` branch, it only understands SMALL_MOTION.
+    # With the `small_motion` branch, it also understands SMALL_MOTION.
     if s == 'CAST':
         return 'BEND'
     elif s == 'STOP_LARGE':