fix(tests): fixes tests