bugfix & unittest