Fixed minor problem with update