Fix code style
Correct method