chore: added tests