Updated to fix build failure