diff --git a/p4/src/main/java/cs448/Project4.java b/p4/src/main/java/cs448/Project4.java
index b3fcfe7..bf9a773 100644
--- a/p4/src/main/java/cs448/Project4.java
+++ b/p4/src/main/java/cs448/Project4.java
@@ -134,7 +134,7 @@ public class Project4 {
 
         Dataset<Row> resultDF = spark.createDataFrame(movieIdRDD, Movie.class);
         resultDF.show();
-        Dataset<String> CompressedResult = resultDF.map((MapFunction<Row, String>) row -> "" + row.getInt(0), Encoders.STRING());
+        Dataset<String> CompressedResult = resultDF.map((MapFunction<Row, String>) row -> "" + row.getInt(1), Encoders.STRING());
         CompressedResult.show();
         CompressedResult.write().text(CS448Utils.resolveUri(conf.outPath, "query-3"));