Fix import path

Signed-off-by: Alexandr Morozov <lk4d4math@gmail.com>
Merge pull request 2