Fix decorator loading bug