Fixed import

Fixed import so that the installation procedure described in README works