Lance Stroll, Bahrain test 2021