Fixing custom edge example.