diff --git a/src/steps/os/freebsd.rs b/src/steps/os/freebsd.rs
index 2c57b77c..67488f62 100644
--- a/src/steps/os/freebsd.rs
+++ b/src/steps/os/freebsd.rs
@@ -1,4 +1,6 @@
 use crate::command::CommandExt;
+use crate::config::Step;
+use crate::execution_context::ExecutionContext;
 use crate::executor::RunType;
 use crate::terminal::print_separator;
 use crate::utils::require_option;
diff --git a/src/steps/upgrade.vim b/src/steps/upgrade.vim
index 28e3df9f..5eac813e 100644
--- a/src/steps/upgrade.vim
+++ b/src/steps/upgrade.vim
@@ -38,6 +38,14 @@ if exists(":CocUpdateSync")
     CocUpdateSync
 endif
 
+" TODO: Should this be after `PackerSync`?
+" Not sure how to sequence this after Packer without doing something weird
+" with that `PackerComplete` autocommand.
+if exists(":TSUpdate")
+    echo "TreeSitter Update"
+    TSUpdate
+endif
+
 if exists(':PackerSync')
   echo "Packer"
   autocmd User PackerComplete quitall