small change and fix