Tests work now