Fixed custom extension class attribute bug