Fixed an issue with output