Fixing post update logic