Tests should work now