Yes, the stewards were pretty consistent in their rulings that if you are the following car and you are pushed off and yet make the pass, you will be penalized if you do not give it back. So Lando and McLaren should have expected the penalty. That being said, Lando pulled out 4.1 seconds by the checkered flag, so I understand why they went for it and hoped they could get more than 5 seconds on Max at the flag.