Fix small change