diff --git a/superset-frontend/spec/javascripts/explore/components/ColorScheme_spec.jsx b/superset-frontend/spec/javascripts/explore/components/ColorScheme_spec.jsx
index b7785198fd0..e41fde91409 100644
--- a/superset-frontend/spec/javascripts/explore/components/ColorScheme_spec.jsx
+++ b/superset-frontend/spec/javascripts/explore/components/ColorScheme_spec.jsx
@@ -39,6 +39,6 @@ describe('ColorSchemeControl', () => {
   });
 
   it('renders a Creatable', () => {
-    expect(wrapper.find(Select)).toExist();
+    expect(wrapper.find(Select)).toHaveLength(1);
   });
 });
diff --git a/tests/charts/api_tests.py b/tests/charts/api_tests.py
index 51ea4bcee13..dbd30e023ac 100644
--- a/tests/charts/api_tests.py
+++ b/tests/charts/api_tests.py
@@ -25,6 +25,7 @@ import prison
 import humanize
 from sqlalchemy.sql import func
 
+from superset.utils.core import get_example_database
 from tests.test_app import app
 from superset.connectors.connector_registry import ConnectorRegistry
 from superset.extensions import db, security_manager