Fixed import os in unittest script.