Fix test imports cause error for pyramid scanning issue