GIRAPH-634: InternalGiraphRunner with TestGraph is broken (majakabiljo)
diff --git a/CHANGELOG b/CHANGELOG
index 98d5490..149eb7e 100644
--- a/CHANGELOG
+++ b/CHANGELOG
@@ -3,6 +3,7 @@
 Release 1.0.1 - unreleased
   
 Release 1.0.0 - 2013-04-12
+  GIRAPH-634: InternalGiraphRunner with TestGraph is broken (majakabiljo)
 
   GIRAPH-630: Convergence detection broken in
   o.a.g.examples.PageRankVertex. (ssc via aching)
diff --git a/giraph-core/src/main/java/org/apache/giraph/utils/InternalVertexRunner.java b/giraph-core/src/main/java/org/apache/giraph/utils/InternalVertexRunner.java
index bb60436..029cb5d 100644
--- a/giraph-core/src/main/java/org/apache/giraph/utils/InternalVertexRunner.java
+++ b/giraph-core/src/main/java/org/apache/giraph/utils/InternalVertexRunner.java
@@ -221,8 +221,10 @@
       File zkMgrDir = FileUtils.createTempDir(tmpDir, "_defaultZkManagerDir");
       File checkpointsDir = FileUtils.createTempDir(tmpDir, "_checkpoints");
 
+      conf.setVertexInputFormatClass(InMemoryVertexInputFormat.class);
+
       // Create and configure the job to run the vertex
-      GiraphJob job = new GiraphJob(conf.getVertexClass().getName());
+      GiraphJob job = new GiraphJob(conf, conf.getVertexClass().getName());
 
       InMemoryVertexInputFormat.setGraph(graph);