Minor fixes to test endpoint