fixed to use custom attr not class