Component (JTable table, Object value, boolean isSelected, int row, int column) { final Component editor = super.getTableCellEditorComponent(table, value, isSelected, row, column); final PsiType type = getRowType(table, row); if (type != null) { ((PsiExpressionCodeFragment)myCodeFragment).setExpectedType(type); } return editor; }