Fix extension error

Fix extension error
Demo

Demo