Fixed problem with os import.