couple quick bug fixs