BUG in 1K dataset. Its not working. Code crashes.