fix bug

Signed-off-by: Philip Miglinci <p.miglinci@gmail.com>