FMS-347 : Fix and add unit test