Issue with Automatic Scaling of RDS Hosts in Azure

Does anyone have a resolution for this issue?
I'm trying to implement this in our Azure environment: https://gallery.technet.microsoft.com/scriptcenter/Automatic-Scaling-of-9b4f5e76/view/Discussions#content
When we run this script against our RD Connection Broker that has multiple collections, we have an issue with all servers in a single collection being set to drain mode. 

Details: We have 2 Collections on the RD Connection Broker. Both Collections have 3 RD Host Servers. The RD Host servers in each Collection are each in their own Resource Group, meaning all RD Host Servers for a collection are grouped into a single resource group.

I set the Config to keep up 2 Servers in off peak mode and run the script against a resource group. All servers in Collection A were put into Drain Mode. All 3 RD Host servers in the collection had a user logged in, so that may have caused an issue as well.

Can anyone offer suggestions on how to force this script to only focus on a single collection? Or is there a better way of accomplishing this?
Thanks in advance!