Fixed compilation errors in tests