Fixed minor error where a NoneType was asked to update