Fixed to avoid deprecation.

Fixes #33.