Put back the change I shouldn't have removed