Fix no output issue