tests: add missing os import