Fixes problem exhibited by ensuring.scala

Need to account for the possibility that function arguments are wrapped in braces. Review by @DarkDimius