Restore removed import