Fix import of os