Trying to fix problem with dataset