[Fix] Correct return values