Fixed some code formatting in previous patch.