Fixes broken test loader