Correct testing logic