Fixed wrong import

@czue