Code

Fixed bug in unit test ;)