Fixed the to set method