Minor bug fix
Minor bug fix