Bug fix
Minor change