Тут проблема вовсе не во flex. Для мобильной верстки обязательно указывай в head вот такую строчку

<meta name="viewport" content="width=device-width, initial-scale=1">

Иначе браузер считает, что есть scrollbar. Кстати проблема проявлялась только со sticky в хроме :)
flex сделан правильно
Про этот метатэг можно почитать здесь