diff --git a/superset/dashboards/api.py b/superset/dashboards/api.py
index 30c0cb3de2c..6d8fd0da555 100644
--- a/superset/dashboards/api.py
+++ b/superset/dashboards/api.py
@@ -89,6 +89,7 @@ from superset.dashboards.schemas import (
     DashboardNativeFiltersConfigUpdateSchema,
     DashboardPostSchema,
     DashboardPutSchema,
+    DashboardScreenshotPostSchema,
     EmbeddedDashboardConfigSchema,
     EmbeddedDashboardResponseSchema,
     get_delete_ids_schema,
@@ -312,6 +313,7 @@ class DashboardRestApi(BaseSupersetModelRestApi):
         TabsPayloadSchema,
         GetFavStarIdsSchema,
         EmbeddedDashboardResponseSchema,
+        DashboardScreenshotPostSchema,
     )
     apispec_parameter_schemas = {
         "get_delete_ids_schema": get_delete_ids_schema,
diff --git a/superset/sqllab/api.py b/superset/sqllab/api.py
index a8ffd92f24f..906dd72bcaf 100644
--- a/superset/sqllab/api.py
+++ b/superset/sqllab/api.py
@@ -88,6 +88,7 @@ class SqlLabRestApi(BaseSupersetApi):
     openapi_spec_component_schemas = (
         EstimateQueryCostSchema,
         ExecutePayloadSchema,
+        FormatQueryPayloadSchema,
         QueryExecutionResponseSchema,
         SQLLabBootstrapSchema,
     )