YAY Fix the visuals