Fix bug in theme