CustomView bug fix