fix attribute error

@maethu