Fix import (travis issue).