Better UI/UX for selecting folder in Android

Currently the dialog for Folder selection is quite hard to use. It only shows less than 2 folders at a time so scrolling through it is slow. Instead, it could be implemented like in the other screenshot of Material Design samples.

Current behaviour:

Suggestion: