Fix small style issues