Whoops :) Clearly we don't actually test `grad` functionality using this function...