BF: Make inputs and outputs from the magic conform in shape.
Merge pull request #2 from arokem/haoxing-magic_test

BF: Make inputs and outputs from the magic conform in shape.