fix units tests