Fix assert. Thanks, Timur!