Fix up issue with last update