Fix Optional import