Fix assert per suggestion