Fix import example in the documentation