small change we forgot to make