bug fixed

add extend