Xiaomi Vacuum Issue

[sheminasalam]

I have been using another custom integration(https://github.com/nqkdev/home-assistant-vacuum-styj02ym) for my xiaomi robovac mop p till recently. But it stopped working with the recent HA update. I then shifted to this which recognizes my vacuum but I am having trouble sending it to clean rooms. Earlier I used to call service like shown below to clean rooms.

But now when I call this service, the vacuum starts cleaning but it is totally disoriented. Then i have to restart the vacuum to make it work normally even with the mobile app, How can I make this work?