-quick logic fix