Forgot to flatten