Fix test failure.

Address #38.