diff --git a/sway/.config/sway/config b/sway/.config/sway/config
index 4407608..06c3eda 100644
--- a/sway/.config/sway/config
+++ b/sway/.config/sway/config
@@ -18,7 +18,7 @@ bindsym Control+Mod1+l exec $lock
 
 bindsym Print exec grim -g "$(slurp -d)" - | wl-copy
 
-bindsym $mod+p exec kitty -e /usr/bin/expect -c "spawn python; expect \\"*>>> \\"; send \\"import numpy as np\n\\"; send \\"import sympy as sp\n\\"; interact"
+bindsym $mod+p exec kitty -e /usr/bin/expect -c "spawn python; expect \\"*>>> \\"; send \\"import numpy as np\n\\"; send \\"import sympy as sp\n\\"; send \\"import torch\n\\"; interact"
 bindsym $mod+Return exec $term
 bindsym $mod+Shift+q kill
 bindsym $mod+d exec $menu