chore: adjust test

It is working now
2 files changed