Как правильно переопределить класс QSqlTableModel?

Я нашел в интернете переопределенный класс tablemodel.h:

class TableModel : public QSqlTableModel
{
    Q_OBJECT
public:
    explicit TableModel(QObject *parent = 0);
    QVariant data(const QModelIndex &idx, int role) const;
};

tablemodel.cpp

#include "tablemodel.h"
#include <QColor>
TableModel::TableModel(QObject *parent) : QSqlTableModel(parent)
{
}
QVariant TableModel::data(const QModelIndex &idx, int role) const
{
    if(role == Qt::BackgroundColorRole){
        if(QSqlTableModel::data(this->index(idx.row(), 0)).toInt() == 6){
            return QColor(Qt::red);
        }
    } else if(role == Qt::DisplayRole){
        return QSqlTableModel::data(idx);
    }
    return QVariant();
}

И все бы хорошо, но в таблице с моделью этого класса при двойном нажатии исчезают данные ячейки, не копируются, ведет себя странно. Как правильно его переопределить его, чтобы работало как с QSqlTableModel?