bug fix. Thanks Daniel Korzekwa!