Fixed a warning
New unit test