tests: add the needed imports