Fixing some errors in tests