Add import lost in merge