Fix if statement code format