Fix import after renaming the lib