Correct problem from a test