tests updated
missing imports