Update test.hs to work with new log2