1.1 --- a/jitcraft/Modules.h	Mon Jul 02 01:24:57 2012 -0700
     1.2 +++ b/jitcraft/Modules.h	Mon Jul 02 01:52:57 2012 -0700
     1.3 @@ -20,6 +20,7 @@
     1.4  
     1.5  #include "Heap.h"
     1.6  #include "Code.h"
     1.7 +#include "Functions.h"
     1.8  
     1.9  namespace jc {
    1.10  
    1.11 @@ -30,13 +31,13 @@
    1.12    public:
    1.13      static Module *New(Zone *zone, unsigned nfunctions);
    1.14  
    1.15 -    Code *getFunction(size_t i) {
    1.16 -        return Code::cast(functions_->at(i));
    1.17 +    Function *getFunction(size_t i) {
    1.18 +        return Function::cast(functions_->at(i));
    1.19      }
    1.20 -    void setFunction(Zone *zone, size_t i, Code *code) {
    1.21 +    void setFunction(Zone *zone, size_t i, Function *fun) {
    1.22          assert(!functions_->at(i));
    1.23 -        assert(code);
    1.24 -        functions_->set(zone, i, code);
    1.25 +        assert(fun);
    1.26 +        functions_->set(zone, i, fun);
    1.27      }
    1.28  
    1.29      static inline Module *cast(Object *obj) {