ER diagram fixes