Minor Bug fix.