diff --git a/extractor-smoke-test/test.sh b/extractor-smoke-test/test.sh
index 387416d8e4d..735f8d1c33c 100755
--- a/extractor-smoke-test/test.sh
+++ b/extractor-smoke-test/test.sh
@@ -8,6 +8,7 @@ cd $DIR
 rm -rf testdb
 
 codeql database create --language=go testdb --search-path ..
+codeql dataset check testdb/db-go
 codeql query run ../ql/test/library-tests/semmle/go/controlflow/ControlFlowGraph/ControlFlowNode_getASuccessor.ql --database=testdb --output=notracing-out.bqrs --search-path ..
 codeql bqrs decode notracing-out.bqrs --format=csv --output=notracing-out.csv
 diff -w -u notracing-out.csv expected.csv
@@ -19,6 +20,7 @@ export CODEQL_EXTRACTOR_GO_BUILD_TRACING=on
 rm -rf testdb
 
 codeql database create --language=go testdb --search-path ..
+codeql dataset check testdb/db-go
 codeql query run ../ql/test/library-tests/semmle/go/controlflow/ControlFlowGraph/ControlFlowNode_getASuccessor.ql --database=testdb --output=tracing-out.bqrs --search-path ..
 codeql bqrs decode tracing-out.bqrs --format=csv --output=tracing-out.csv
 diff -w -u tracing-out.csv expected.csv