Jamie Vardy is one of four Leicester players one booking away from a suspension, and manager Craig Shakespeare has urged his side to keep their discipline against Atletico Madrid.