Lab2 fix