@fix to work inside class