Fix rest kit integration, not relevant to the test but avoids build errors