Fix graph parameter and test