Updated to prevent breaking things
Minor update