Fix IDE integration (wrong return type)