diff --git a/com.ibm.wala.cast/source/java/com/ibm/wala/cast/loader/AstFunctionClass.java b/com.ibm.wala.cast/source/java/com/ibm/wala/cast/loader/AstFunctionClass.java
index 4c7e54a9e..b148e99c3 100644
--- a/com.ibm.wala.cast/source/java/com/ibm/wala/cast/loader/AstFunctionClass.java
+++ b/com.ibm.wala.cast/source/java/com/ibm/wala/cast/loader/AstFunctionClass.java
@@ -32,7 +32,7 @@ abstract public class AstFunctionClass implements IClass, ClassConstants {
   
   private final IClassLoader loader;
 
-  protected IMethod functionBody;
+  protected AstMethod functionBody;
 
   private final CAstSourcePositionMap.Position sourcePosition;
 
@@ -217,7 +217,7 @@ abstract public class AstFunctionClass implements IClass, ClassConstants {
     return true;
   }
   
-  public IMethod getCodeBody() {
+  public AstMethod getCodeBody() {
     return functionBody;
   }
 }