Changed setup to fix error