Fix unit tests of framework