Bug fix

Signed-off-by: Mida Boghetich <mida@boghetich.com>