fix framework failure in regression test

@xiang90 @xiaoyunwu 
Xiang is correct.

Xiaoyun, since this is for review only, you can just open an issue/gist and point me (and Xiang) to the diffs you are concerned about instead of send a PR.

IMHO, we should clean things up in and send only those changes  that matter. :)