Minor logic correction