JZ: Fix failing viewset test