Fix takeWhile'