Updated tests and code to work with latest release of drf.