Fix set method