1 files changed, 5 insertions, 4 deletions
diff --git a/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerRepository.java b/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerRepository.java
index b613ff8..53a62be 100644
--- a/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerRepository.java
+++ b/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerRepository.java
@@ -1,16 +1,17 @@
 package org.fedoraproject.candlepin.model;
 
-import org.hibernate.Session;
+import javax.persistence.EntityManager;
+
 import org.hibernate.criterion.Restrictions;
 
 public class ConsumerRepository extends AbstractHibernateRepository<Consumer> {
     
-    public ConsumerRepository(Session session) {
-        super(session);
+    public ConsumerRepository(EntityManager em) {
+        super(em);
     }
 
     public Consumer lookupByName(String name) {
-        return (Consumer) session.createCriteria(Consumer.class)
+        return (Consumer) currentSession().createCriteria(Consumer.class)
             .add(Restrictions.like("name", name))
             .uniqueResult();
     }