refactor a tiny bit

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