Fix the vi-mode widget when vi-mode is disabled