bug fix - add lock