Minor bugfixes

Needed import and fixed a semi-colon