Fix issue where output was not being returned back into template fixed #1

Fix the issue where the plugin did not return the modified output.