Quick fix for compilation errors