minor fix to main file