Fix missing instrumentation