Minor error fix