Fixed wrong return type