Fixed the edge bug
Fixed the edge bug