FIX: added missing os import