small error fix to main