fix import

Signed-off-by: Ricky Elrod <ricky@elrod.me>