small update to fix error