diff --git a/src/key-viewer-dialog.hpp b/src/key-viewer-dialog.hpp
index 60996af..01261a5 100644
--- a/src/key-viewer-dialog.hpp
+++ b/src/key-viewer-dialog.hpp
@@ -21,10 +21,11 @@
 #define NDN_NCC_KEY_VIEWER_DIALOG_HPP
 
 #include "key-tree-model.hpp"
+#include "cert-tree-model.hpp"
 #include <ndn-cxx/security/key-chain.hpp>
 
 #include <QDialog>
-#include <QStandardItemModel>
+
 
 namespace Ui {
 class KeyViewerDialog;
@@ -76,7 +77,7 @@
   shared_ptr<KeyTreeModel> m_model;
   shared_ptr<KeyChain> m_keyChain;
 
-  shared_ptr<QStandardItemModel> m_tableModel;
+  shared_ptr<CertTreeModel> m_tableModel;
 };
 
 } // namespace ncc