[TASK] Fix failing unit tests for LocalDriver