Update per @borjasalguero review

Signed-off-by: Rick Waldron <waldron.rick@gmail.com>