Use the correct import path