Minor bug fixing.