Tests fix for in body

Fix for #2