diff --git a/create_fork.sh b/create_fork.sh
index f4766ef..4d147c3 100755
--- a/create_fork.sh
+++ b/create_fork.sh
@@ -76,8 +76,8 @@ cp -i calculate_average_$SOURCE_FORK.sh calculate_average_$FORK.sh
 substitute_in_file $SOURCE_FORK $FORK calculate_average_$FORK.sh
 
 if [ $SOURCE_FORK == "baseline" ]; then
-  cp -i src/main/java/dev/morling/onebrc/CalculateAverage.java src/main/java/dev/morling/onebrc/CalculateAverage_$FORK.java
-  substitute_in_file CalculateAverage CalculateAverage_$FORK src/main/java/dev/morling/onebrc/CalculateAverage_$FORK.java
+  cp -i src/main/java/dev/morling/onebrc/CalculateAverage_baseline.java src/main/java/dev/morling/onebrc/CalculateAverage_$FORK.java
+  substitute_in_file CalculateAverage_baseline CalculateAverage_$FORK src/main/java/dev/morling/onebrc/CalculateAverage_$FORK.java
 else
   cp -i src/main/java/dev/morling/onebrc/CalculateAverage_$SOURCE_FORK.java src/main/java/dev/morling/onebrc/CalculateAverage_$FORK.java
   substitute_in_file $SOURCE_FORK $FORK src/main/java/dev/morling/onebrc/CalculateAverage_$FORK.java