Update units tests