@@ -54,7 +54,7 @@ async def handle_list_prompts(ctx: RequestContext) -> list[types.Prompt]:
 )
 from mcp.shared.version import SUPPORTED_PROTOCOL_VERSIONS
 
-from src.mcp.server.transport_session import TransportSession
+from mcp.server.transport_session import TransportSession
 
 
 class InitializationState(Enum):
Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ async def handle_list_prompts(ctx: RequestContext) -> list[types.Prompt]:`
`54`	`54`	`)`
`55`	`55`	`from mcp.shared.version import SUPPORTED_PROTOCOL_VERSIONS`
`56`	`56`
`57`		`-from src.mcp.server.transport_session import TransportSession`
	`57`	`+from mcp.server.transport_session import TransportSession`
`58`	`58`
`59`	`59`
`60`	`60`	`class InitializationState(Enum):`