fix error, very close to make it work now