diff --git a/extensions/persist/test/com/google/inject/persist/jpa/EnsureJpaCanTakeObjectsInPropertiesTest.java b/extensions/persist/test/com/google/inject/persist/jpa/EnsureJpaCanTakeObjectsInPropertiesTest.java
index c5421a5..6dcd6bd 100644
--- a/extensions/persist/test/com/google/inject/persist/jpa/EnsureJpaCanTakeObjectsInPropertiesTest.java
+++ b/extensions/persist/test/com/google/inject/persist/jpa/EnsureJpaCanTakeObjectsInPropertiesTest.java
@@ -77,7 +77,7 @@
     injector.getInstance(EntityManagerFactory.class).close();
   }
 
-  private static JDBCDataSource getDataSource() {
+  private static DataSource getDataSource() {
     final JDBCDataSource dataSource = new JDBCDataSource();
     dataSource.setDatabase("jdbc:hsqldb:mem:persistence");
     dataSource.setUser("sa");
@@ -86,14 +86,13 @@
   }
 
   private void startPersistService(boolean passDataSource) {
-    final JDBCDataSource dataSource = getDataSource();
+    final DataSource dataSource = getDataSource();
 
     injector = Guice.createInjector(new DBModule(dataSource, passDataSource));
 
     //startup persistence
     injector.getInstance(PersistService.class).start();
   }
-
   public void testWorksIfPassDataSource() {
     startPersistService(true);
   }