Fix instrumentation for rails 4