fix unit tests after changes in framework