fix oneshot diff problem