Correct logic
Correct logic