Fixes bug on command line argument