Small fix visibility fix