fix(test): Fix the tests!