I needed these changes to make it work