Small fix, forgot to change before initial commit