fix for code review