Fix test errors.
Merge pull request #43 from prabhuramachandran/fix_test_error

Fix test errors.