Fixing references to get it to compile.