Do you mean the returning of the original function if DEBUG is True?  That was in the original code so I just left it in.