fix the bug in the modification of the problem