This logic is correct