Minor update
Minor patch for an Issue #404