Removed incorrect condition
Added test data