Issue #4: Small bug fix.