fixed bug from prev update
Signed-off-by: Brian Coca <briancoca+dev@gmail.com>