Stylecop fix

Fix StyleCop warnings
This is weird now. Probably just add curly braces instead.