Fixed missing imports

Signed-off-by: fhenry <felix.henry@sopra.com>