Fix for bug

Signed-off-by: Ryan Benoit <rdbenoit@me.com>