Fix bug for CI