Fix bug for small changes