fixed some small bug

Signed-off-by: Nicholas Tan Jerome <nicholas.jerome@kit.edu>