Fixed missing import in command.