Fix: Invalid import statements in tests

Signed-off-by: Kevin Conway <kevinjacobconway@gmail.com>