Fix DRF 2 and 3 compat