Fix the state parameter implementation