Fixed the problematic assert