Fix bug with command line argument