Fix import + remove wrong test #56