fixing bug, I forgot to return