Update DrupalContext.php

return $row that we are looking for instead of the whole $element