Fix #18

Fixes a mysterious import that slipped in #18 and removes an unused import.

Didn't expect for it to be merged before the Travis build went through.