Copy/paste error and small logic change.