add missing include; fix copy/paste wrong variable name

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