Fix import path (thanks Diogo Franco)