Fixing test input, missing observation ids