Small fix + add reference