fixed bug in extension case