Set correct class for the icons

Set correct class for the icons