diff --git a/nodes/src/main/java/org/nodes/models/ERSimpleModel.java b/nodes/src/main/java/org/nodes/models/ERSimpleModel.java
index 932cdb8..6d4472f 100644
--- a/nodes/src/main/java/org/nodes/models/ERSimpleModel.java
+++ b/nodes/src/main/java/org/nodes/models/ERSimpleModel.java
@@ -49,7 +49,7 @@ public double codelength(Graph<? extends Object> graph)
 		}
 	}
 	
-	public static double undirected(int size, int numLinks, boolean withPrior)
+	public static double undirected(long size, long numLinks, boolean withPrior)
 	{
 		double n = size;
 		double t = n * (n - 1) / 2;
@@ -57,7 +57,7 @@ public static double undirected(int size, int numLinks, boolean withPrior)
 	}
 	
 	
-	public static double directed(int size, int numLinks, boolean withPrior)
+	public static double directed(long size, long numLinks, boolean withPrior)
 	{
 		double n = size;
 		double t = n * n - n;