forgot to update the header

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