1 files changed, 3 insertions, 9 deletions
diff --git a/commands/term-close.go b/commands/term-close.go
index 38fcc27..30f3392 100644
--- a/commands/term-close.go
+++ b/commands/term-close.go
@@ -3,7 +3,6 @@ package commands
 import (
 	"errors"
 
-	"git.sr.ht/~sircmpwn/aerc2/lib/ui"
 	"git.sr.ht/~sircmpwn/aerc2/widgets"
 )
 
@@ -15,15 +14,10 @@ func TermClose(aerc *widgets.Aerc, args []string) error {
 	if len(args) != 1 {
 		return errors.New("Usage: term-close")
 	}
-	grid, ok := aerc.SelectedTab().(*ui.Grid)
+	thost, ok := aerc.SelectedTab().(*widgets.TermHost)
 	if !ok {
 		return errors.New("Error: not a terminal")
 	}
-	for _, child := range grid.Children() {
-		if term, ok := child.(*widgets.Terminal); ok {
-			term.Close(nil)
-			return nil
-		}
-	}
-	return errors.New("Error: not a terminal")
+	thost.Terminal().Close(nil)
+	return nil
 }