DB lab2 edge bug fix