adapt unit tests