Add workaround for travis tests