Forgot to add these minor changes to use the icon