Minor logical fix