Fix the setStyle not working issue