diff --git a/include/cppkafka/topic_partition.h b/include/cppkafka/topic_partition.h
index 97ddc919..3176d4d8 100644
--- a/include/cppkafka/topic_partition.h
+++ b/include/cppkafka/topic_partition.h
@@ -106,6 +106,11 @@ class CPPKAFKA_API TopicPartition {
      * Gets the offset
      */
     int64_t get_offset() const;
+    
+    /**
+     * @brief Sets the partition
+     */
+    void set_partition(int partition);
 
     /**
      * Sets the offset
diff --git a/src/topic_partition.cpp b/src/topic_partition.cpp
index 8ded9ee0..35cab420 100644
--- a/src/topic_partition.cpp
+++ b/src/topic_partition.cpp
@@ -76,6 +76,10 @@ int64_t TopicPartition::get_offset() const {
     return offset_;
 }
 
+void TopicPartition::set_partition(int partition) {
+    partition_ = partition;
+}
+
 void TopicPartition::set_offset(int64_t offset) {
     offset_ = offset;
 }