Fix check for number of command line arguments