diff --git a/src/QueryBuilder/QueryBuilder/TestQuery/TestQuery.js b/src/QueryBuilder/QueryBuilder/TestQuery/TestQuery.js
index 5801b695..0a4fbef2 100644
--- a/src/QueryBuilder/QueryBuilder/TestQuery/TestQuery.js
+++ b/src/QueryBuilder/QueryBuilder/TestQuery/TestQuery.js
@@ -198,6 +198,7 @@ TestQuery.propTypes = {
   onQueryRetrieved: PropTypes.func,
   onQueryExecutionSuccess: PropTypes.func,
   onQueryExecutionFail: PropTypes.func,
+  onSetDefaultVisibleColumns: PropTypes.func,
   testQuery: PropTypes.func,
   isTestQueryLoading: PropTypes.bool,
   isPreviewLoading: PropTypes.bool,
@@ -205,4 +206,5 @@ TestQuery.propTypes = {
   isTestQueryInProgress: PropTypes.bool,
   setIsTestQueryInProgress: PropTypes.func,
   recordsLimit: PropTypes.number,
+  recordColumns: PropTypes.arrayOf(PropTypes.string)
 };