Fix extension example
Fix missing import for extensions