We should be doing this for pure Python defined classe as well,
but don't have a way to inject it.