simplifying one_shot