fix(tests): start fixing unit tests