Fix wrong return type