Minor ML visual fix