Fix tests running with DRF 2.4