Fix runtime bug with custom cell class.