Fix the output issues

By fixing the references