Fix compilation error caused by wrong imports