Fixed bok choy flaky test for studio.

TNL-284

On ajax call ```wait_for_page()``` was used instead of ```wait_for_ajax()``` in ```add_missing_groups()```
No special reason, it looks odd there with ```wait_for_ajax()```.