Fixing the UI damage from errorous commit.