Allow to set custom Reference class