Fix missing assert