Update aspect to track method argument