Fixed tests to avoid compilation errors