Added a missing use clause in test

Signed-off-by: Ivan Habunek <ivan.habunek@gmail.com>