Bugfix

Needed to import library earlier