diff --git a/langgraph-go/advancedgraph/graph.go b/langgraph-go/advancedgraph/graph.go
index 1ab681fb8..d0d8c5177 100644
--- a/langgraph-go/advancedgraph/graph.go
+++ b/langgraph-go/advancedgraph/graph.go
@@ -108,7 +108,7 @@ func (h *Handler) WaitForResult() (map[string]any, error) {
 	return res.state, res.err
 }
 
-func (g *CompiledGraph) Start(initialState map[string]any, initialInput any) (*Handler, error) {
+func (g *CompiledGraph) Start(initialInput any, initialState map[string]any) (*Handler, error) {
 	engine := NewRustEngine()
 	for _, ch := range g.asyncChannels {
 		if err := engine.AddAsyncChannel(ch); err != nil {
diff --git a/langgraph-go/tests/sub_agents_test.go b/langgraph-go/tests/sub_agents_test.go
index a03c16523..c660bddf4 100644
--- a/langgraph-go/tests/sub_agents_test.go
+++ b/langgraph-go/tests/sub_agents_test.go
@@ -174,12 +174,12 @@ func TestSubAgentsEquivalentFlow(t *testing.T) {
 	graph.SetFinishNode(workflow.orderFoodNode)
 
 	handler, err := graph.Compile().Start(
+		nil,
 		map[string]any{
 			"input":  "help me get something for lunch",
 			"output": []string{},
 			"done":   nil,
 		},
-		nil,
 	)
 	if err != nil {
 		t.Fatalf("start failed: %v", err)
diff --git a/langgraph-go/tests/test_primitives_test.go b/langgraph-go/tests/test_primitives_test.go
index b58b75fc8..70bf4a091 100644
--- a/langgraph-go/tests/test_primitives_test.go
+++ b/langgraph-go/tests/test_primitives_test.go
@@ -71,10 +71,10 @@ func TestInputAndStatePrimitivesCompatible(t *testing.T) {
 	graph.AddNode(workflow.middleNode)
 	graph.AddFinishNode(workflow.finishNode)
 
-	handler, err := graph.Compile().Start(map[string]any{
+	handler, err := graph.Compile().Start(100, map[string]any{
 		"logs": []string{},
 		"done": nil,
-	}, 100)
+	})
 	if err != nil {
 		t.Fatalf("start failed: %v", err)
 	}