Fixing test + forgotten imports

Signed-off-by: Damien DUPORTAL <damien.duportal@gmail.com>
Gofmt

Signed-off-by: Damien DUPORTAL <damien.duportal@gmail.com>