minor fix to main