STY: Made test more stringent. Removed .loc[] reassignment.