fix broken tests.
Merge pull request #1 from astj/fix_test

fix broken tests.