Fix use statement.

Closes #266.