Fix missing header & wrong return type