After quietly pausing the rollout of Google Photos’ AI-powered “Ask Photos” search tool, Google is now expanding access once again and making some improvements to the feature.

Google’s Gemini AI models power Ask Photos so that you can ask complex questions to help you find photos. But earlier this month, a member of the Google Photos team said on X that the feature “isn’t where it needs to be, in terms of latency, quality and UX.”

In a blog post published Thursday, Google said that it has “heard your feedback” that the feature should “return more photos faster for simple searches, like ‘beach’ or ‘dogs.’” Now, “you’ll now see results right away while Gemini models continue to work in the background to find the most relevant photos or information for more complex queries,” according to Google.

The company adds that the feature is now “opening up beyond early access” and is beginning to roll out to more “eligible users” in the US.