Minor code changes.

Signed-off-by: Mithron <mithron@gmail.com>