Slight error fix