Small visual fix..