diff --git a/ludwig/serve.py b/ludwig/serve.py
index 6896b8ac4f1..831477134dd 100644
--- a/ludwig/serve.py
+++ b/ludwig/serve.py
@@ -41,8 +41,6 @@
     from starlette.middleware import Middleware
     from starlette.middleware.cors import CORSMiddleware
     from starlette.requests import Request
-
-    # from starlette.responses import JSONResponse
 except ImportError as e:
     logger.error(e)
     logger.error(
diff --git a/tests/ludwig/utils/test_server_utils.py b/tests/ludwig/utils/test_server_utils.py
index 9c93779706c..34d64700527 100644
--- a/tests/ludwig/utils/test_server_utils.py
+++ b/tests/ludwig/utils/test_server_utils.py
@@ -1,5 +1,4 @@
 import numpy as np
-import pytest
 
 from ludwig.utils.server_utils import NumpyJSONResponse