Fix for poor test