Fix import path 

Related to #14