Fixed the damage upgrade... hopefully!