Update test implementation to extend ExecutionContext