Correctly remove the internal theano trace. As we return only 1 level to the user, this made us return useless information.