Fix custom style not removed