Updated tests to reflect usage in bug

USERGRID-1960