Minor bug fix

.