Fixed a null pointer exception in  the test