diff --git a/cookbook/mock_prompt_management_server/mock_prompt_management_server.py b/cookbook/mock_prompt_management_server/mock_prompt_management_server.py
index 8f00e0a5ed..295a96e12a 100644
--- a/cookbook/mock_prompt_management_server/mock_prompt_management_server.py
+++ b/cookbook/mock_prompt_management_server/mock_prompt_management_server.py
@@ -18,7 +18,7 @@ import os
 import json
 from typing import Any, Dict, List, Optional
 
-from fastapi import FastAPI, HTTPException, Header, Query
+from fastapi import FastAPI, HTTPException, Header, Query, status
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel, Field