Fix failing travis test