diff --git a/src/commands/command_p.h b/src/commands/command_p.h
--- a/src/commands/command_p.h
+++ b/src/commands/command_p.h
@@ -37,6 +37,7 @@
 #include "view/keylistcontroller.h"
 
 #include <Libkleo/KeyListModel>
+#include <Libkleo/KeyRearrangeColumnsProxyModel>
 
 #include <KMessageBox>
 
@@ -73,7 +74,7 @@
     }
     KeyListModelInterface *model() const
     {
-        return view_ ? dynamic_cast<KeyListModelInterface *>(view_->model()) : nullptr;
+        return view_ ? dynamic_cast<KeyListModelInterface *>(dynamic_cast<KeyRearrangeColumnsProxyModel*>(view_->model())) : nullptr;
     }
     KeyListController *controller() const
     {