fix(tests): fix broken unit tests