This wasn't working for me, simple fix instead

Sorry, totally brain-dead from tiredness!