fixing similar bug

for input edge