Fix bug in prepping data