diff --git a/main.py b/main.py
index 6ebc238..acb2a61 100644
--- a/main.py
+++ b/main.py
@@ -28,14 +28,16 @@ async def media_info(pk: int) -> Media:
     return cl.media_info(pk)
 
 
-@app.get("/", tags=["system"])
+@app.get("/", tags=["system"], summary="Redirect to /docs")
 async def root():
+    """Redirect to /docs
+    """
     return RedirectResponse(url="/docs")
 
 
-@app.get("/version", tags=["system"])
+@app.get("/version", tags=["system"], summary="Get dependency versions")
 async def version():
-    """Return package versions
+    """Get dependency versions
     """
     versions = {}
     for name in ('instagrapi', ):
diff --git a/tests.py b/tests.py
index daafda7..be6430d 100644
--- a/tests.py
+++ b/tests.py
@@ -24,7 +24,7 @@ async def test_media_info() -> None:
         )
     media = response.json()
     assert response.status_code == 200
-    assert media["pk"] == "2110901750722920960"
+    assert media["pk"] == 2110901750722920960
     assert media["id"] == "2110901750722920960_8572539084"
     assert media["code"] == "B1LbfVPlwIA"
     assert media["media_type"] == 1