Fix prob2 bug