Fixed bad import, fixes #9