Fix reference and documentation