Minor bug fix, cosmetics.