1 files changed, 2 insertions, 0 deletions
diff --git a/decoder/sampler.h b/decoder/sampler.h
index e5840f41..4d0b2e64 100644
--- a/decoder/sampler.h
+++ b/decoder/sampler.h
@@ -88,10 +88,12 @@ typedef RandomNumberGenerator<boost::mt19937> MT19937;
 class SampleSet {
  public:
   const prob_t& operator[](int i) const { return m_scores[i]; }
+  prob_t& operator[](int i) { return m_scores[i]; }
   bool empty() const { return m_scores.empty(); }
   void add(const prob_t& s) { m_scores.push_back(s); }
   void clear() { m_scores.clear(); }
   size_t size() const { return m_scores.size(); }
+  void resize(int size) { m_scores.resize(size); }
   std::vector<prob_t> m_scores;
 };