Fixes #4.

Just forgot to import.