Fixes refactor issue with display_result