diff --git a/profile.ps1 b/profile.ps1
index de501cd..160b847 100644
--- a/profile.ps1
+++ b/profile.ps1
@@ -67,6 +67,8 @@ Set-PSReadLineKeyHandler -Chord "Ctrl+o" -Function ForwardWord
 # Change cursor shape based on what Vi mode the line editor is in.
 # Tested and works with PowerShell 5.1
 # NOTE: Requires: Install-Module PsReadline -Scope CurrentUser -Force
+# TODO: This doesn't handle all cases well. For example when exiting terminal
+# nvim in a PowerShell session the cursor is a block while in insert mode.
 function OnViModeChange {
   if ($args[0] -eq 'Command') {
     # Set the cursor to non-blinking block.