Fix: Must cast the return type here