Fixing bug on milestone graph representation