fix import and remove debugging