diff --git a/extras/src/test/java/edu/umd/cs/piccolox/pswing/PSwingTest.java b/extras/src/test/java/edu/umd/cs/piccolox/pswing/PSwingTest.java
index d8a3cac..3ff96d0 100644
--- a/extras/src/test/java/edu/umd/cs/piccolox/pswing/PSwingTest.java
+++ b/extras/src/test/java/edu/umd/cs/piccolox/pswing/PSwingTest.java
@@ -38,6 +38,8 @@
 import javax.swing.JPanel;
 import javax.swing.RepaintManager;
 
+import edu.umd.cs.piccolo.util.PPaintContext;
+
 import junit.framework.TestCase;
 
 /**
@@ -100,8 +102,8 @@
 
         final BufferedImage img = new BufferedImage(100, 100, BufferedImage.TYPE_INT_RGB);
         final Graphics2D graphics = GraphicsEnvironment.getLocalGraphicsEnvironment().createGraphics(img);
-
-        pSwing.paint(graphics);
+        final PPaintContext paintContext = new PPaintContext(graphics);;
+        pSwing.paintComponent(paintContext);
         assertEquals(Color.RED.getRGB(), img.getRGB(50, 50));
     }