hopefully fix the sampling