View extension bug fix