Fixed test compilation errors