Fix the management command