fix bug

Signed-off-by: James Rubingh <james@wrive.com>