Visual bug fix