cadence-workflow
diff --git a/‎src/main/java/com/uber/cadence/internal/sync/TestWorkflowEnvironmentInternal.java
Lines changed: 2 additions & 2 deletions b/‎src/main/java/com/uber/cadence/internal/sync/TestWorkflowEnvironmentInternal.java
Lines changed: 2 additions & 2 deletions
@@ -498,7 +498,7 @@ public void DeprecateDomain(
     public void RestartWorkflowExecution(
         RestartWorkflowExecutionRequest restartRequest, AsyncMethodCallback resultHandler)
         throws TException {
-      impl.RestartWorkflowExecution(restartRequest);
+      impl.RestartWorkflowExecution(restartRequest, resultHandler);
     }
 
     @Override
@@ -743,7 +743,7 @@ public void CountWorkflowExecutions(
 
     @Override
     public void GetSearchAttributes(AsyncMethodCallback resultHandler) throws TException {
-      impl.GetSearchAttributes();
+      impl.GetSearchAttributes(resultHandler);
     }
 
     @Override
Original file line number	Diff line number	Diff line change
`@@ -498,7 +498,7 @@ public void DeprecateDomain(`
`498`	`498`	`public void RestartWorkflowExecution(`
`499`	`499`	`RestartWorkflowExecutionRequest restartRequest, AsyncMethodCallback resultHandler)`
`500`	`500`	`throws TException {`
`501`		`- impl.RestartWorkflowExecution(restartRequest);`
	`501`	`+ impl.RestartWorkflowExecution(restartRequest, resultHandler);`
`502`	`502`	`}`
`503`	`503`
`504`	`504`	`@Override`
`@@ -743,7 +743,7 @@ public void CountWorkflowExecutions(`
`743`	`743`
`744`	`744`	`@Override`
`745`	`745`	`public void GetSearchAttributes(AsyncMethodCallback resultHandler) throws TException {`
`746`		`- impl.GetSearchAttributes();`
	`746`	`+ impl.GetSearchAttributes(resultHandler);`
`747`	`747`	`}`
`748`	`748`
`749`	`749`	`@Override`