fixed one smal issue