[fix] Fix assertion