2 simple changes (2)

2 simple changes (2)