Fixes invalid tests