Import error fixes