That's probably the case, but I think it's a gradual progression. The funeral happened shortly after their return, but when he later visited Hurley at Santa Rosa (the first time), he was fine. Between the guilt of Claire, raising Aaron, and the visions of his father (and knowing Hurley is having similar visions about Charlie) slowly breaks him down to the point where we saw him in "Through the Looking Glass."