Fixes import bug in server tests

Signed-off-by: Oren Leiman <oren@neutral.io>