Small visual fix