Fix JS Unit test