Missed some setting through review

Added some setting code.