More coverage. Upgrade to drapers #source_class instead of model_class