Fix for leftover test