fix for custom  class