Update JSON test to check for model type in response