Small bug fix curly brace...