fix: added tests