Fix import errors for docs