diff --git a/src/foodwebs/omnivory.jl b/src/foodwebs/omnivory.jl
index 9f6ee28e1..43b291ba6 100644
--- a/src/foodwebs/omnivory.jl
+++ b/src/foodwebs/omnivory.jl
@@ -1,8 +1,21 @@
+"""
+    omnivory(N::T) where {T <: UnipartiteNetwork}
+
+Returns a vector of length |species(N)| with an index of consumption across
+trophic levels. Note we explicitly assume that diet contribution is spread
+evenly across prey species.
+
+#### References
+- Christensen, Villy, and Daniel Pauly. "ECOPATH II—a software for balancing
+  steady-state ecosystem models and calculating network characteristics."
+  Ecological modelling 61, no. 3-4 (1992): 169-185.
+
+"""
 function omnivory(N::T) where {T <: UnipartiteNetwork}
     OI = Dict([s => 0.0 for s in species(N)])
 
-    TL = fractional_trophic_level(N)
-    k = degree(N; dims=1)
+    TL = trophic_level(N)
+    k = degree_in(N)
 
     for sp_i in species(N)
 
@@ -11,10 +24,10 @@ function omnivory(N::T) where {T <: UnipartiteNetwork}
 
         # For every species, we set its initial omnivory to 0
         oi = 0.0
-        for (j, sp_j) in enumerate(species(N))
+        for sp_j in species(N)
             # Then for every species it consumes, we ha
             tl_diff = (TL[sp_j] - (TL[sp_i]-1.0)).^2.0
-            corr = N[sp_i,sp_j]/k[sp_i]
+            corr = N[sp_j,sp_i]/k[sp_i]
             oi += tl_diff * corr
         end
         OI[sp_i] = oi