Missing input error, when calling traceplot() on a loaded trace

I am trying to call traceplot() on a model trace loaded from disk, like so:

trace = pm.load_trace(directory = {directory})
pm.traceplot(trace)

It yields a missing input error:

TypeError: Missing required input: {var_name} log_

The loaded trace does have an underlying array and I am able to use it with sample_posterior_predictive(trace).

Am I missing something?