[T4] Fix unit tests