fixed the border issue.

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