Fix bug

Signed-off-by: Gabriel Reiser <greiser@cardfree.com>