fix(test): unit tests fix