Fix compilation errors

now compiles fine......