fixed bug

Signed-off-by: Michael Maloof <michaelmaloof1991@gmail.com>