Fixing bug. Forget to update command.py