fixed little logical bug