Fix update functionality to maintain association on update
Release v0.3.2