Minor code error fix