Fix compilation errors in some classes