fix test
Merge pull request #8 from BrandyMint/vadus/feature/fix_test

fix test