DOC: Fix import