Updated test.

Fixes #4
Upgraded some stuff.

Fixes #2