Fix forgetting bug