This needs to be made into a graph partition for test to run properly