Fixing a deprecated set call