chore: fix for tests