chore: fix failing unit tests