Found some minor errors in tests