fixed import error in test/Main.hs