fix extend fn
fix extend fn