fixing small error in bug-assignment code