Fix bug in main styling