fixed the displaying issues by adding an import