Put border on correct container this time