retool wait_condition to use count attribute

Change-Id: I6c681757ce89336e1a61f952fb4c655ad76ff1bd