Minor logic update.