Fix to schema. Forgot foreign key..