Fix small bug, border works as expected now.