Remove incorrect import from docs example