Update output
Bug fix.