Seems to be working fine for me. I have test1 and test2 folders on my bookmarks bar. I goto a site, click the heart icon and select the test2 bookmarks bar folder. Then, I goto another site and click the heart icon and test2 is already selected by default.

If I then bookmark another page to test1, and then bookmark yet another page, it does in test1 automatically.