fixing mistake made from last commit