Small fix, otherwise it won't work