missing import

Signed-off-by: Steven Armstrong <steven@icarus.ethz.ch>