Fix dependency import