Highlighted moves when in check are now correct.