Fix for test to pass build