Naomi Osaka recovered after a mid-match lapse to advance to the third round of the US Open with a 6-2, 5-7, 6-1 victory over Katerina Siniakova.