Fix input test bugs