ENH: fix logic setup