fix(tests): Fixes unit testing.