Added correctness check