Adding statement to import os to two test