Fix wrong type hints, closes #1

This also fixes a test that was previously doing nothing.

Found with eastwood.