fix(test): fixed unit tests