fix(tests): Fixes tests.