Checkpoint
Fixed module input test breaks.