small fix for first review