Warning fixed. Test updated