Minor test fix.
Merge pull request #352 from tonyhuang/master

Minor test fix.