Robert feature fix

fixed hopefully