Fixing missing os import