I forgot to maintain the unit tests! I fixed them