Add import fix