diff --git a/ludwig/datasets/configs/imdb_genre_prediction.yaml b/ludwig/datasets/configs/imdb_genre_prediction.yaml
index 7244bb4efc4..19fb8b94fa7 100644
--- a/ludwig/datasets/configs/imdb_genre_prediction.yaml
+++ b/ludwig/datasets/configs/imdb_genre_prediction.yaml
@@ -27,7 +27,9 @@ columns:
     type: category
   - name: Runtime (Minutes)
     type: number
-  - name: Rating Votes
+  - name: Rating
+    type: Number
+  - name: Votes
     type: number
   - name: Revenue (Millions)
     type: number