fixed the import and reference

Signed-off-by: Morgan Harold Bauer <bauer.morgan@gmail.com>