Fix TestGraph to be generic Collection