Add forgotten os import